package com.topten.beans;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.topten.basicas.varejo.Estoque;

public class EstoqueDataModel extends ListDataModel<Estoque> implements
		SelectableDataModel<Estoque> {

	public EstoqueDataModel() {
	}

	public EstoqueDataModel(List<Estoque> data) {
		super(data);
	}

	@Override
	public Estoque getRowData(String arg0) {

		List<Estoque> itensEstoque = (List<Estoque>) getWrappedData();
	
		for (Estoque estoque : itensEstoque) {
			if (estoque.getCodigo().toString().equals(arg0)) {

				return estoque;

			}

		}

		return null;
	}

	@Override
	public Object getRowKey(Estoque estoque) {
		
		return estoque.getProduto().getDescricao();
	}

}
